#ifndef __NET_H__
#define __NET_H__

void conv2d_f32(float* dout, float* din, int din_hgt, int din_wid, const float* ker, const float* bias, const int* shape); 
void fc_32(float* dout, float* din, const float* weight, const float* bias, const int* shape);
void relu(float* dout, float* din, int size);
void relu2d(float* dout, float* din, int num_c, int size);
void maxpool2d(float* dout, float* din, int din_hgt, int din_wid, int num_c, int ksize);
void flatten(float* dout, float* din, int out_len, int in_size, int num_c);

#endif
